package com.lzy.canal.cilent;

import com.alibaba.otter.canal.client.CanalConnector;
import com.alibaba.otter.canal.client.CanalConnectors;
import com.alibaba.otter.canal.protocol.CanalEntry.*;
import com.alibaba.otter.canal.protocol.Message;
import com.google.protobuf.InvalidProtocolBufferException;
import org.apache.commons.dbutils.DbUtils;
import org.apache.commons.dbutils.QueryRunner;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import javax.sql.DataSource;
import java.net.InetSocketAddress;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.Iterator;
import java.util.List;
import java.util.Queue;
import java.util.concurrent.ConcurrentLinkedQueue;

/**
 * @ClassName:   CanalClient
 * @Author:      Li ZhenYong
 * @Date:        2021/12/26 15:35
 * @Version:  1.0
 */
@Component
public class CanalClient {

	//sql队列
	//下面判断拼接后的sql会加入这个队列
	private Queue<String> SQL_QUEUE = new ConcurrentLinkedQueue<>();

	@Resource
	private DataSource dataSource;

	/**
	 * canal入库方法
	 */
	//指定要监听的ip地址的canal端口号，默认开启为11111
	public void run() {
		CanalConnector connector = CanalConnectors.newSingleConnector(new
				InetSocketAddress("47.110.247.184",
				11111), "example", "", "");
		int batchSize = 1000;
		try {
			connector.connect();
			connector.subscribe(".*\\..*");
			connector.rollback();
			try {
				while (true) {
					//尝试从master那边拉去数据batchSize条记录，有多少取多少
					//监控上面ip的数据库是否变化
					Message message = connector.getWithoutAck(batchSize);
					long batchId = message.getId();
					int size = message.getEntries().size();
					if (batchId == -1 || size == 0) {
						//没变化就睡
						Thread.sleep(1000);
					} else {
						//有变化就同步
						dataHandle(message.getEntries());
					}
					connector.ack(batchId);
					//当队列里面堆积的sql大于一定数值的时候就模拟执行
					//队列里如果有sql语句就执行
					if (SQL_QUEUE.size() >= 1) {
						executeQueueSql();
					}
				}
			} catch (InterruptedException e) {
				e.printStackTrace();
			} catch (InvalidProtocolBufferException e) {
				e.printStackTrace();
			}
		} finally {
			connector.disconnect();
		}
	}

	/**
	 * 模拟执行队列里面的sql语句
	 */
	public void executeQueueSql() {
		int size = SQL_QUEUE.size();
		for (int i = 0; i < size; i++) {
			String sql = SQL_QUEUE.poll();
			System.out.println("[sql]----> " + sql);
			this.execute(sql.toString());
		}
	}

	/**
	 * 数据处理
	 *
	 * @param entrys
	 */
	private void dataHandle(List<Entry> entrys) throws
			InvalidProtocolBufferException {
		for (Entry entry : entrys) {
			if (EntryType.ROWDATA == entry.getEntryType()) {
				RowChange rowChange = RowChange.parseFrom(entry.getStoreValue());
				EventType eventType = rowChange.getEventType();
				//判断当前是什么操作：删除、更新、插入
				if (eventType == EventType.DELETE) {
					saveDeleteSql(entry);
				} else if (eventType == EventType.UPDATE) {
					saveUpdateSql(entry);
				} else if (eventType == EventType.INSERT) {
					saveInsertSql(entry);
				}
			}
		}
	}


	//保存更新语句
	private void saveUpdateSql(Entry entry) {
		try {
			RowChange rowChange = RowChange.parseFrom(entry.getStoreValue());
			List<RowData> rowDatasList = rowChange.getRowDatasList();
			for (RowData rowData : rowDatasList) {
				List<Column> newColumnList = rowData.getAfterColumnsList();
				StringBuffer sql = new StringBuffer("update " +
						entry.getHeader().getTableName() + " set ");
				for (int i = 0; i < newColumnList.size(); i++) {
					sql.append(" " + newColumnList.get(i).getName()
							+ " = '" + newColumnList.get(i).getValue() + "'");
					if (i != newColumnList.size() - 1) {
						sql.append(",");
					}
				}
				sql.append(" where ");
				List<Column> oldColumnList = rowData.getBeforeColumnsList();
				for (Column column : oldColumnList) {
					if (column.getIsKey()) {
						//暂时只支持单一主键
						sql.append(column.getName() + "=" + column.getValue());
						break;
					}
				}
				SQL_QUEUE.add(sql.toString());
			}
		} catch (InvalidProtocolBufferException e) {
			e.printStackTrace();
		}
	}

	//保存删除语句
	private void saveDeleteSql(Entry entry) {
		try {
			RowChange rowChange = RowChange.parseFrom(entry.getStoreValue());
			List<RowData> rowDatasList = rowChange.getRowDatasList();
			for (RowData rowData : rowDatasList) {
				List<Column> columnList = rowData.getBeforeColumnsList();
				StringBuffer sql = new StringBuffer("delete from " +
						entry.getHeader().getTableName() + " where ");
				for (Column column : columnList) {
					if (column.getIsKey()) {
						//暂时只支持单一主键
						sql.append(column.getName() + "=" + column.getValue());
						break;
					}
				}
				SQL_QUEUE.add(sql.toString());
			}
		} catch (InvalidProtocolBufferException e) {
			e.printStackTrace();
		}
	}


	//保存插入语句
	private void saveInsertSql(Entry entry) {
		try {
			RowChange rowChange = RowChange.parseFrom(entry.getStoreValue());
			List<RowData> rowDatasList = rowChange.getRowDatasList();
			for (RowData rowData : rowDatasList) {
				List<Column> columnList = rowData.getAfterColumnsList();
				StringBuffer sql = new StringBuffer("insert into " +
						entry.getHeader().getTableName() + " (");
				for (int i = 0; i < columnList.size(); i++) {
					sql.append(columnList.get(i).getName());
					if (i != columnList.size() - 1) {
						sql.append(",");
					}
				}
				sql.append(") VALUES (");
				for (int i = 0; i < columnList.size(); i++) {
					sql.append("'" + columnList.get(i).getValue() + "'");
					if (i != columnList.size() - 1) {
						sql.append(",");
					}
				}
				sql.append(")");
				SQL_QUEUE.add(sql.toString());
			}
		} catch (InvalidProtocolBufferException e) {
			e.printStackTrace();
		}
	}

	//入库
	//与数据库连接，并执行队列里面取出的sql语句
	public void execute(String sql) {
		Connection con = null;
		try {
			if (null == sql) return;
			con = dataSource.getConnection();
			QueryRunner qr = new QueryRunner();
			int row = qr.execute(con, sql);
			System.out.println("update: " + row);
		} catch (SQLException e) {
			e.printStackTrace();
		} finally {
			DbUtils.closeQuietly(con);
		}
	}


}

